# 导包
import requests
# 创建测试类（模块）
class TestIHRMLogin:
    # 登录成功
    def test01_login_success(self):
        url = "http://ihrm-test.itheima.net/api/sys/login"
        login_data = {
            "mobile": "13800000002",
            "password": "123456"
        }
        response = requests.post(url=url, json=login_data)
        print(response.json())

        # 断言响应状态码
        assert response.status_code == 200
        # 断言json返回数据中的success值
        assert response.json().get("success") is True
        # 断言json返回数据中的code值
        assert response.json().get("code") == 10000
        # 断言返回数据中包含指定消息message
        assert "成功" in response.text

    # 登录失败（手机号错误）
    def test02_mobile_error(self):
        url = "http://ihrm-test.itheima.net/api/sys/login"
        login_data = {
            "mobile": "",
            "password": "123456"
        }
        response = requests.post(url=url, json=login_data)
        print(response.json())

        # 断言响应状态码
        assert response.status_code == 200
        # 断言json返回数据中的success值
        assert response.json().get("success") is False
        # 断言json返回数据中的code值
        assert response.json().get("code") == 20001
        # 断言返回数据中包含指定消息message
        assert "错误" in response.text

    # 登录失败（密码错误）
    def test03_password_error(self):
        url = "http://ihrm-test.itheima.net/api/sys/login"
        login_data = {
            "mobile": "13800000002",
            "password": "123457"
        }
        response = requests.post(url=url, json=login_data)
        print(response.json())

        # 断言响应状态码
        assert response.status_code == 200
        # 断言json返回数据中的success值
        assert response.json().get("success") is False
        # 断言json返回数据中的code值
        assert response.json().get("code") == 20001
        # 断言返回数据中包含指定消息message
        assert "错误" in response.text
